# 导包
import requests
from commom.ihrm_login import ihrm_login
from config import host


# 1-定义类（1个模块）
class IhrmUserApi:
    # 2-定义类属性
    user_url = host + "/api/sys/user/"
    add_url = host + "/api/sys/user"
    head = {"Authorization": ihrm_login()}

    # 添加员工
    @classmethod
    def ihrm_user_add(cls, add_json):  # 对于数量变化的数据，可以整体作为一个参数传递
        # 4-执行接口请求
        head = {"Authorization": ihrm_login()}  # 定义请求头
        resp = requests.post(url=cls.add_url, json=add_json, headers=head)
        print(resp.json())
        # 5-返回响应对象
        return resp

    # 删除员工
    @classmethod
    def ihrm_user_delete(cls, user_id):
        # 执行接口请求
        req_url = cls.user_url + user_id  # 拼接请求URL
        resp = requests.delete(url=req_url, headers=cls.head)
        # 返回响应结果
        return resp
